#include "BlockQueue.hpp"
#include "Task.hpp"

void* Consumer(void* args)
{
    BlockQueue<Task>* bq = static_cast<BlockQueue<Task>*>(args);
    while(true)
    {
        // 消费
        Task t = bq->pop();
        // t.GetResult();
        // t.run();
        t();
        std::cout << "处理任务: " << t.GetTask() << ", 运算结果是: " << t.GetResult() << ", thread id: " << pthread_self() << std::endl;
        // sleep(1);
    }
}

void* Product(void* args)
{
    int len = opers.size();
    BlockQueue<Task>* bq = static_cast<BlockQueue<Task>*>(args);
    // int x = 1;
    // int y = 1;
    while(true)
    {
        int data1 = rand() % 10 + 1;
        int data2 = rand() % 5 + 1;
        char op = opers[rand() % len];

        // sleep(2);
        // 生产
        Task t(data1, data2, op);
        bq->push(t);
        // std::cout << "生产了一个任务: " << t.GetTask() << ", thread id: " << pthread_self() << std::endl;
        printf("生产了一个任务：%s, th=hread id : %d\n", t.GetTask(), pthread_self());
        sleep(1);
    }
}

int main()
{
    srand(time(nullptr));
    BlockQueue<Task>* bq = new BlockQueue<Task>;
    pthread_t c[3],p[5];
    for(int i = 0;i < 3; ++i)
    {
        pthread_create(c + i, nullptr, Consumer, bq);
    }

    for(int i = 0; i < 5; ++i)
    {
        pthread_create(p + i, nullptr, Product, bq);
    }

    for(int i = 0;i < 3; ++i)
    {
        pthread_join(c[i],nullptr);
    }

    for(int i = 0;i < 5; ++i)
    {
        pthread_join(p[i],nullptr);
    }

    delete bq;
    return 0;
}